How can I change zero matrix to one according to the predetermined row and column in python